package com.errorfinding.rabbitmq.four;

import cn.hutool.log.Log;
import com.errorfinding.rabbitmq.util.RabbitMqUtils;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;

import java.io.IOException;
import java.util.UUID;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;
import java.util.concurrent.TimeoutException;

/**
 * 发布确认模式
 * 使用的时间 比较哪种确认模式是是最好的
 * 1.单个确认模式
 * 2.批量确认模式
 * 3.异步批量确认模式
 */
public class ConfirmMessage {

    //批量发消息的个数
    public static final int messageCount = 1000;

    public static void main(String[] args) throws IOException, InterruptedException, TimeoutException {
        //单个确认
        //ConfirmMessage.publishMessageIndividually(); // 发布1000个单独确认消息,耗时：26878ms
        //批量确认
        //ConfirmMessage.publishMessageBatch(); //发布1000个批量确认消息,耗时：324ms
        //异步确认
        //ConfirmMessage.publishMessageAsync(); //发布1000个异步确认消息,耗时：68ms

    }

    //单个确认
    public static void publishMessageIndividually() throws IOException, TimeoutException, InterruptedException {
        //获取信道
        Channel channel = RabbitMqUtils.getChannel();
        //队列声明
        String queueName = UUID.randomUUID().toString(); //队列名称
        channel.queueDeclare(queueName,true,false,false,null);
        //开启发布确认模式
        channel.confirmSelect();
        //开始时间
        long begin = System.currentTimeMillis();

        //批量发布
        for (int i = 0; i < messageCount; i++) {
            String message = i+" ";
            //发送消息
            channel.basicPublish("",queueName,null,message.getBytes());
            //单个消息就马上进行发布确认
            //channel.waitForConfirms(); 等待确认发布
            channel.waitForConfirms();
        }
        //结束时间
        long end = System.currentTimeMillis();
        //用时
        System.out.println("发布"+messageCount+"个单独确认消息,耗时："+(end-begin)+"ms");
    }

    //批量发布确认
    public static void publishMessageBatch() throws IOException, InterruptedException, TimeoutException {
        //获取信道
        Channel channel = RabbitMqUtils.getChannel();
        //队列声明
        String queueName = UUID.randomUUID().toString(); //队列名称
        channel.queueDeclare(queueName,true,false,false,null);
        //开启发布确认模式
        channel.confirmSelect();
        //开始时间
        long begin = System.currentTimeMillis();
        //批量确认消息大小
        int batchSize = 100;

        //批量发布消息,批量发布确认
        for (int i = 0; i < messageCount; i++) {
            String message = i+"";
            channel.basicPublish("",queueName,null,message.getBytes());
            //判断达到100条消息的时候 确认一次
            if (i%batchSize == 0){
                //确认发布
                channel.waitForConfirms();
            }
        }
        //结束时间
        long end = System.currentTimeMillis();
        //用时
        System.out.println("发布"+messageCount+"个批量确认消息,耗时："+(end-begin)+"ms");
    }

    //异步确认发布
    public static void publishMessageAsync() throws IOException, TimeoutException {
        //获取信道
        Channel channel = RabbitMqUtils.getChannel();
        //队列声明
        String queueName = UUID.randomUUID().toString(); //队列名称
        channel.queueDeclare(queueName,true,false,false,null);
        //开启发布确认模式
        channel.confirmSelect();

        /**
         * 线程安全有序的一个哈希表，使用于高并发的情况下
         * 1.轻松的将序号与消息进行关联
         * 2.轻松批量删除条目 只要给到序号
         * 3.支持高并发(多线程)
         */
        ConcurrentSkipListMap<Long,String> concurrentSkipListMap = new ConcurrentSkipListMap<>();

        //开始时间
        long begin = System.currentTimeMillis();

        //消息确认成功回调函数
        ConfirmCallback ackCallback = (deliveryTag,multiple)->{
            //2.删除掉已经确认的消息 剩下的就是未确认的消息
            if (multiple){
                ConcurrentNavigableMap<Long,String> confirmed = concurrentSkipListMap.headMap(deliveryTag,true);
                confirmed.clear();
            }else {
                concurrentSkipListMap.remove(deliveryTag);
            }

            System.out.println("确认的消息："+deliveryTag);
        };

        //消息确认失败回调函数
        /**
         * 参数1。消息的标记
         * 参数2。是否为批量确认
         */
        ConfirmCallback nackCallback = (deliveryTag,multiple)->{
            //3.打印一下未确认的消息有哪些
            String nackMessage = concurrentSkipListMap.get(deliveryTag);

            System.out.println("未确认的消息："+deliveryTag);
        };

        //准备消息的监听器 监听哪些消息成功了 哪些消息失败了
        /**
         * 参数1。监听哪些消息成功了
         * 参数2。哪些消息失败了
         */
        channel.addConfirmListener(ackCallback,nackCallback); //异步通知

        //批量发布消息
        for (int i = 0; i < messageCount; i++) {
            String message = "消息"+i;
            channel.basicPublish("",queueName,null,message.getBytes());
            //1.此处记录下所有要发送的消息 消息的总和
            concurrentSkipListMap.put(channel.getNextPublishSeqNo(),message);
        }

        //结束时间
        long end = System.currentTimeMillis();
        //用时
        System.out.println("发布"+messageCount+"个异步确认消息,耗时："+(end-begin)+"ms");
    }
}
